import os
"""
 使用文件和目录操作，定义一个统计指定目录大小的函数（注意目录中还有子目录）。
"""


# 定义指定目录大小的函数
def count_dir(path):
    # 初始的大小为0
    num = 0
    # 如果是文件就可以直接计算文件的大小
    if os.path.isfile(path):
        num += os.path.getsize(path)
    # 判断是否是目录
    if os.path.isdir(path):
        # 获取指定文件的内容
        list_dir = os.listdir(path)
        # 遍历所有的文件
        for List in list_dir:
            file = os.path.join(path, List)
            if os.path.isdir(file):
                num = num + count_dir(file)  # 递归调用

            if os.path.isfile(file):
                num += os.path.getsize(file)

    return num


print(count_dir('./'))
